package leetcode;

import java.util.ArrayList;
import java.util.List;

/**
 * 在英语中，我们有一个叫做 词根(root) 的概念，可以词根后面添加其他一些词组成另一个较长的单词——我们称这个词为 继承词(successor)。例如，词根an，跟随着单词 other(其他)，可以形成新的单词 another(另一个)。
 *
 * 现在，给定一个由许多词根组成的词典 dictionary 和一个用空格分隔单词形成的句子 sentence。你需要将句子中的所有继承词用词根替换掉。如果继承词有许多可以形成它的词根，则用最短的词根替换它。
 *
 * 你需要输出替换之后的句子。
 *
 *
 *
 * 示例 1：
 *
 * 输入：dictionary = ["cat","bat","rat"], sentence = "the cattle was rattled by the battery"
 * 输出："the cat was rat by the bat"
 *
 * 示例 2：
 *
 * 输入：dictionary = ["a","b","c"], sentence = "aadsfasf absbs bbab cadsfafs"
 * 输出："a a b c"
 *
 *
 *
 * 提示：
 *
 *     1 <= dictionary.length <= 1000
 *     1 <= dictionary[i].length <= 100
 *     dictionary[i] 仅由小写字母组成。
 *     1 <= sentence.length <= 10^6
 *     sentence 仅由小写字母和空格组成。
 *     sentence 中单词的总量在范围 [1, 1000] 内。
 *     sentence 中每个单词的长度在范围 [1, 1000] 内。
 *     sentence 中单词之间由一个空格隔开。
 *     sentence 没有前导或尾随空格。
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/replace-words
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */

public class WordReplace {
    public static void main(String[] args) {
        WordReplace wordReplace = new WordReplace();
        List<String> list = new ArrayList<>();
        String sentence = "aadsfasf absbs bbab cadsfafs";
        list.add("cat");
        list.add("bat");
        list.add("rat");


        System.out.println("final result: "+ wordReplace.replaceWords(list,sentence));
    }

    static class TreeNode{
        String word;
        TreeNode[] children;

        public TreeNode(){
            word = null;
            children = new TreeNode[26];
        }
    }

    public String replaceWords(List<String> dict,String sentence){
        TreeNode treeNode = new TreeNode();
        for(String root:dict){
            TreeNode cur = treeNode;
            for(char letter:root.toCharArray()){
                if(cur.children[letter-'a']==null){
                    cur.children[letter-'a']= new TreeNode();
                }

                cur = cur.children[letter - 'a'];
            }
            cur.word = root;
        }

        StringBuilder ans = new StringBuilder();
        for(String word : sentence.split("\\s+")){
            if(ans.length()>0) {
                ans.append(" ");
            }

            TreeNode cur = treeNode;
            for(char letter:word.toCharArray()){
                if(cur.children[letter-'a'] == null || cur.word != null){
                    break;
                }
                cur = cur.children[letter-'a'];
            }

            ans.append(cur.word != null?cur.word:word);
        }

        return ans.toString();
    }

}
